package com.taiic.deepseek.rkllm;

public class RKLLMResult {
    // Generated text result
    private String text;

    // ID of the generated token
    private int tokenId;

    // Hidden states of the last layer (if requested)
    private RKLLMResultLastHiddenLayer lastHiddenLayer;

    // Constructor
    public RKLLMResult(String text, int tokenId, RKLLMResultLastHiddenLayer lastHiddenLayer) {
        this.text = text;
        this.tokenId = tokenId;
        this.lastHiddenLayer = lastHiddenLayer;
    }

    // Getters and Setters
    public String getText() {
        return text;
    }

    public void setText(String text) {
        this.text = text;
    }

    public int getTokenId() {
        return tokenId;
    }

    public void setTokenId(int tokenId) {
        this.tokenId = tokenId;
    }

    public RKLLMResultLastHiddenLayer getLastHiddenLayer() {
        return lastHiddenLayer;
    }

    public void setLastHiddenLayer(RKLLMResultLastHiddenLayer lastHiddenLayer) {
        this.lastHiddenLayer = lastHiddenLayer;
    }

    @Override
    public String toString() {
        return "RKLLMResult{" +
                "text='" + text + '\'' +
                ", tokenId=" + tokenId +
                ", lastHiddenLayer=" + lastHiddenLayer +
                '}';
    }
}
